#!/usr/bin/env bash

data_dir="sample_path"
model='gpt2'
learning_rate=1e-5
warmup_steps=500
weight_decay=0
batch_size=64
num_epoch=10
save_dir="./checkpoints/"
